import os

from matplotlib import pyplot as plt
import cv2
import pandas as pd
# 我的电脑因为保存图片横纵可能相反 所以在别的电脑上运行可能会出BUG

# 下面的一个数组就会保存一个视频采样图片数据下所有采样图片在采样间隔下计算出的下层污泥高度
def heightCal(path,flag):


    # 高度绘图横坐标
    xaixs = []
    HeightRate1 = []
    XaixsSpeed = []
    ChenSpeed1 = []

    data = pd.read_excel(path)
    for i in range(0, len(data['ASM']) - 1):
        xaixs.append(data['xaixs'][i])
        HeightRate1.append(data['HeightRate1'][i])
        XaixsSpeed.append(data['XaixsSpeed'][i])
        ChenSpeed1.append(data['ChenSpeed1'][i])


    font1 = {'family': 'Times New Roman', 'weight': 'normal', 'size': 20}
    if flag==0:
        plt.figure(figsize=(30, 15))
        plt.plot(xaixs, HeightRate1, linewidth=6)
        legendName = ['AAO-1', 'AAOYuan', 'YangHua1', 'YangHuaYuan']
        font1 = {'family': 'Times New Roman', 'weight': 'normal', 'size': 35}
        font2 = {'family': 'Times New Roman', 'weight': 'normal', 'size': 14}
        plt.tick_params(labelsize=25)
        plt.title('Height Trendency', fontsize=35, pad=40)
        #plt.legend(legendName, prop=font1)
        plt.xlabel('time/min', font1)
        plt.ylabel('height/Ratio', font1)
        plt.grid(True)
        plt.show()

    # 绘图 高度变化趋势

        print("执行完了")
    if flag==1:

    # 绘图 高度变化速度趋势
        plt.figure(figsize=(30, 15))
        plt.plot(XaixsSpeed, ChenSpeed1, linewidth=6)
        legendName = ['AAO-1', 'AAOYuan', 'YangHua1', 'YangHuaYuan']
        font1 = {'family': 'Times New Roman', 'weight': 'normal', 'size': 35}
        font2 = {'family': 'Times New Roman', 'weight': 'normal', 'size': 14}
        plt.tick_params(labelsize=25)
        plt.title('precipitate speed', fontsize=35, pad=40)
        #plt.legend(legendName, prop=font1)
        plt.xlabel('time/min', font1)
        plt.ylabel('height/%', font1)
        plt.grid(True)
        plt.show()
